
import argparse
from image_diff_detector import ImageDifferenceDetector

def parse_arguments():
    parser = argparse.ArgumentParser(description='图像差异检测工具')
    parser.add_argument('--img1', required=True, help='第一张图片路径')
    parser.add_argument('--img2', required=True, help='第二张图片路径')
    parser.add_argument('--lang', default='ch', help='OCR 语言（默认中文）')
    parser.add_argument('--output_json', default='diff_log.json', help='输出差异日志 JSON')
    parser.add_argument('--output_image', default='diff_result.png', help='输出差异图像文件')
    return parser.parse_args()

def main():
    args = parse_arguments()
    detector = ImageDifferenceDetector(lang=args.lang)
    log = detector.compare_images(
        img1_path=args.img1,
        img2_path=args.img2,
        output_json=args.output_json,
        output_image=args.output_image
    )

    print(f"检测完成，共发现差异 {len(log)} 处。")
    print(f"差异日志已保存至: {args.output_json}")
    print(f"差异图像已保存至: {args.output_image}")

if __name__ == '__main__':
    main()
